package openapi

const (
	DEEPSEEK = 1
	TONGYI   = 2
	GEMINI   = 3
)

// ModelProperties API提供商属性
type ModelProperties struct {
	Agent string // API基础URL
	Token string // API访问令牌
}

// GetModelProperties 根据提供商类型获取模型属性
func GetModelProperties(agent int) ModelProperties {
	switch agent {
	case DEEPSEEK:
		return ModelProperties{
			Agent: "https://api.deepseek.com/v1",
			Token: "sk-a12fcd16ec2849af95e0a34f2235d7ea",
		}
	case TONGYI:
		return ModelProperties{
			Agent: "https://dashscope.aliyuncs.com/compatible-mode/v1",
			Token: "sk-fd3aae1390f04092a6f7f34baf0b2709", // 请替换为实际的Token
		}
	case GEMINI:
		return ModelProperties{
			Agent: "https://generativelanguage.googleapis.com/v1beta/openai/",
			Token: "AIzaSyA5yLtSFygw_wUmNbaaroLP9680qSckFnc", // 请替换为实际的Token
		}
	default:
		return ModelProperties{
			Agent: "https://api.deepseek.com/v1", // 默认使用Deepseek
			Token: "sk-a12fcd16ec2849af95e0a34f2235d7ea",
		}
	}
}
